09. TensorDataset 和批处理数据

8 TensorDataset 批处理 V1

被忽略的步骤:重排数据

记得重排数据,这样模型就不会学习数据的顺序,而是专注于内容。我们可以设置 shuffle=True 并通过数据加载器重排数据。练习和 solution notebook 已经更新了代码。

# make sure to SHUFFLE your data
train_loader = DataLoader(train_data, shuffle=True, batch_size=batch_size)
valid_loader = DataLoader(valid_data, shuffle=True, batch_size=batch_size)
test_loader = DataLoader(test_data, shuffle=True, batch_size=batch_size)

TensorDataset

查看 TensorDataset 类的代码后会发现,它的作用是提供 easy way to create a dataset out of standard data structures